package com.powernode.config;

import com.powernode.model.SwaggerProperties;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment;
import springfox.documentation.builders.RequestHandlerSelectors;
import springfox.documentation.service.ApiInfo;
import springfox.documentation.service.Contact;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spring.web.plugins.Docket;

import javax.annotation.Resource;
import java.util.HashSet;

/**
 * swagger 自动配置类
 *
 * @author devhjb
 */
@Configuration
@EnableConfigurationProperties(SwaggerProperties.class)
public class SwaggerAutoConfiguration {

    @Resource
    private SwaggerProperties swaggerProperties;

    /**
     * 项目环境对象
     */
    @Resource
    private Environment environment;

    /**
     * 设置 swagger 文档对象
     * swagger文档只在开发阶段生成 , 生产环境不需要生成文档
     * OAS_30 = new DocumentationType("openApi", "3.0");
     *
     * @return
     */
    @Bean
    public Docket docket() {
        // 定义决定是否开启生成api文档的 开关
        boolean flag = true;
        // 判断是否是 生产环境 , 生产环境不生产api文档
        String[] activeProfiles = environment.getActiveProfiles();
        for (String activeProfile : activeProfiles) {
            if (activeProfile.equals("pro")) {
                // 是生产环境
                flag = false;
                break;
            }
        }

        // 从项目环境
        return new Docket(DocumentationType.OAS_30) // 指定为OAS_30 , 即api为3.0版本
                .apiInfo(getApiInfo()) // 获取文档描述信息对象信息
                .enable(flag)   // 是否生成api文档的开关
                .select()
                .apis(RequestHandlerSelectors.basePackage(swaggerProperties.getBasePackage()))  // 基于包名描述swagger注解
                .build();
    }

    /**
     * 获取存储到 swaggerProperties模型中的信息
     *
     * @return
     */
    private ApiInfo getApiInfo() {
        Contact contact = new Contact(
                swaggerProperties.getName(),
                swaggerProperties.getUrl(),
                swaggerProperties.getEmail());
        return new ApiInfo(
                swaggerProperties.getTitle(),
                swaggerProperties.getDescription(),
                swaggerProperties.getVersion(),
                swaggerProperties.getTermsOfServiceUrl(),
                contact,
                swaggerProperties.getLicense(),
                swaggerProperties.getLicenseUrl(),
                new HashSet<>());
    }


}
